# -*- coding: UTF-8 -*-
"""
@File    ：5.动态GDP.py
@Author  ：techPang
@Date    ：2025/2/5 16:21 
@Version ：1.0
"""
from pyecharts.charts import Bar, Timeline
from pyecharts.globals import ThemeType
from pyecharts.options import LabelOpts, TitleOpts

f = open("./1960-2019全球GDP数据.csv", "r", encoding="GB2312")
# 获取文件内容，转为一个列表，每一行内容作为列表的一个元素
data_lines = f.readlines()
# 删除第一个元素
data_lines.pop(0)
# 将数据转为字典存储，year为字典，country和gdp为值
data_dict = {}
for line in data_lines:
    year = int(line.split(",")[0])
    country = line.split(",")[1]
    gdp = float(line.split(",")[2])
    try:
        # 追加数据
        data_dict[year].append([country, gdp])
    except KeyError:
        # 如果抛出异常，先创建一个空的，之后再添加
        data_dict[year] = []
        data_dict[year].append([country, gdp])

timeline = Timeline({"theme": ThemeType.LIGHT})
# 排序年份，年份升序
sorted_year_list = sorted(data_dict.keys())
for year in sorted_year_list:
    # 对列表进行排序，降序
    data_dict[year].sort(key=lambda element: element[1], reverse=True)
    # 取出本年份前8名的国家
    year_data = data_dict[year][0:8]
    x_data = []
    y_data = []
    for country_gdp in year_data:
        x_data.append(country_gdp[0])
        y_data.append(country_gdp[1] / 100000000)

    # 构建柱状图
    bar = Bar()
    x_data.reverse()
    y_data.reverse()
    bar.add_xaxis(x_data)
    bar.add_yaxis("GDP(亿)", y_data, label_opts=LabelOpts(position="right"))
    # 反转x轴和y轴
    bar.reversal_axis()
    # 设置每一年的标题
    bar.set_global_opts(title_opts=TitleOpts(title=f"{year}全球前8GDP数据"))
    timeline.add(bar, str(year))

# 设置时间线自动播放
timeline.add_schema(
    play_interval=1000,  # 自动播放间隔
    is_timeline_show=True,  # 是否在自动播放的时候，显示时间线
    is_auto_play=True,  # 是否自动播放
    is_loop_play=False  # 是否循环播放
)

# 绘图
timeline.render("1960-2019全球GDP前8国家.html")
